NumPy-1: 다차원 배열과 인덱싱

참조 자료

NumPy는 과학 컴퓨팅을 위한 핵심 라이브러리로, 다차원 배열 연산 및 선형대수, 난수 생성 등의 기능을 제공합니다.

import numpy as np
X = np.array([0.1, 0.2, 0.3])
print(X.shape)  # (3,)
print(X.dtype)  # float64
print(X.ndim)   # 1차원

Y = np.zeros([3,3])  # 모든 요소를 0으로 초기화
Y = np.ones([3,3])   # 모든 요소를 1로 초기화

A = np.array([])
for i in range(3):
    A = np.append(A, [1,2])
print(A)  # [1. 2. 1. 2. 1. 2.]

np.arange(1, 1.5, 0.1)  # [1.  1.1  1.2  1.3  1.4]
np.linspace(0., 20., 5)  # [ 0.  5. 10. 15. 20.]

X = np.arange(9)
Y = X.reshape([3,3])
print(Y)

A = np.array([1.2, -1.3, 2.2, 5.3, 3.7])
print(A[0])  # 1.2
print(A[0:3])  # [1.2, -1.3, 2.2]

X = np.array([1.2, -1.3, 2.2, 5.3, 3.7])
print(X[X > 2])  # [2.2, 5.3, 3.7]

A = np.array([[1, 2], [3, 4]])
B = np.array([[5, 6]])
np.concatenate((A, B), axis=0)  # 행 방향 합치기
np.concatenate((A, B.T), axis=1)  # 열 방향 합치기